package socket;
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 聊天室的服务器3
 将消息发送给所有客户端
 */
public class Server4 {
    private ServerSocket serverSocket;

    //该集合用于保存所有客户端的输出流，用于广播消息
    private List<PrintWriter> allOut=new ArrayList<>();

    public Server4(){
        try {
            System.out.println("正在启动服务器");
            serverSocket=new ServerSocket(8088);
            System.out.println("服务端启动完毕");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public  void start(){
        try {
            while (true){
                System.out.println("等待客户端连接...");
                Socket socket=serverSocket.accept();
                System.out.println("一个客户端连接上了!");
                //启动一个线程负责与该客户端交互
                Runnable handler=new ClientHandler(socket);
                Thread t=new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args){
        Server4 server=new Server4();
        server.start();
    }

    //该线程负责与指定的客户端交互
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String host;//记录该客户端的IP地址

        public ClientHandler(Socket socket){
            this.socket=socket;
            //通过socket获取远端计算机的IP地址的字符串格式
            host=socket.getInetAddress().getHostAddress();
        }

        public void run(){
            try {
                InputStream in=socket.getInputStream();
                InputStreamReader isr=new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br=new BufferedReader(isr);


                OutputStream out=socket.getOutputStream();
                OutputStreamWriter osw=new OutputStreamWriter(out,StandardCharsets.UTF_8);
                BufferedWriter bw=new BufferedWriter(osw);
                PrintWriter pw=new PrintWriter(bw,true);

                //将对应该客户端的输出流存入到共享集合allOut中,便于其它线程可以访问
                allOut.add(pw);

                String message;
                while ((message=br.readLine())!=null){
                    System.out.println(host+":"+message);
                    //遍历allOut集合，给所有客户端发送这个消息
                    for(PrintWriter o:allOut){
                        o.println(host+":"+message);
                    }

                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

}
